package com.ruoyi.ade.llm.model;

import java.util.HashMap;
import java.util.Map;

/**
 *   /api/generate
 */
public class CompletionRequest {

    //模型名称
    private String model;
    //提示词
    private String prompt;
    //非流式
    private boolean stream = false;

    private Map<String,Object> options;

    public CompletionRequest(String model, String prompt) {
        this.model = model;
        this.prompt = prompt;
        options = new HashMap<>();
        options.put("temperature",0.1);
    }

    public String getModel() {
        return model;
    }

    public void setModel(String model) {
        this.model = model;
    }

    public String getPrompt() {
        return prompt;
    }

    public void setPrompt(String prompt) {
        this.prompt = prompt;
    }

    public boolean isStream() {
        return stream;
    }

    public void setStream(boolean stream) {
        this.stream = stream;
    }

    public Map<String, Object> getOptions() {
        return options;
    }

    public void setOptions(Map<String, Object> options) {
        this.options = options;
    }
}
